package com.cg.leetcode;

import org.junit.Test;

import java.util.*;

/**
 * 394.字符串解码
 *
 * @program: LeetCode->LeetCode_394
 * @description: 394.字符串解码
 * @author: cg
 * @create: 2021-09-20 21:23
 **/
public class LeetCode_394 {

    @Test
    public void test394() {
        System.out.println(decodeString("3[a]2[bc]"));
        System.out.println(decodeString("abc3[cd]xyz"));
        System.out.println(decodeString("3[a2[c]]"));
    }

    /**
     * 给定一个经过编码的字符串，返回它解码后的字符串。
     * 编码规则为: k[encoded_string]，表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。
     * 你可以认为输入字符串总是有效的；输入字符串中没有额外的空格，且输入的方括号总是符合格式要求的。
     * 此外，你可以认为原始数据不包含数字，所有的数字只表示重复的次数 k ，例如不会出现像 3a 或 2[4] 的输入。
     * <p>
     * 示例 1：
     * 输入：s = "3[a]2[bc]"
     * 输出："aaabcbc"
     * <p>
     * 示例 2：
     * 输入：s = "3[a2[c]]"
     * 输出："accaccacc"
     * <p>
     * 示例 3：
     * 输入：s = "2[abc]3[cd]ef"
     * 输出："abcabccdcdcdef"
     * <p>
     * 示例 4：
     * 输入：s = "abc3[cd]xyz"
     * 输出："abccdcdcdxyz"
     *
     * @param s
     * @return
     */
    public String decodeString(String s) {
        //双栈实现
        StringBuffer res = new StringBuffer();
        Stack<Integer> multiStack = new Stack<>();
        Stack<StringBuffer> ansStack = new Stack<>();
        int multi = 0;
        for (char c : s.toCharArray()) {
            if (Character.isDigit(c)) {
                multi = multi * 10 + c - '0';
            } else if (c == '[') {
                ansStack.push(res);
                multiStack.push(multi);
                res = new StringBuffer();
                multi = 0;
            } else if (Character.isAlphabetic(c)) {
                res.append(c);
            } else {
                StringBuffer lastRes = ansStack.pop();
                int curMulti = multiStack.pop();
                for (int i = 0; i < curMulti; i++) {
                    lastRes.append(res);
                }
                res = lastRes;
            }
        }
        return res.toString();
    }
    /*public String decodeString(String s) {
        //单栈实现
        Deque<String> deque = new LinkedList<>();
        StringBuilder builder;
        for (char c : s.toCharArray()) {
            //将不是]的字符全部进栈
            if (c != ']') {
                deque.push(String.valueOf(c));
            } else {
                builder = new StringBuilder();
                //将[]中的字母出栈
                while (!deque.isEmpty() && deque.peek().matches("[a-z]*")) {
                    builder.insert(0, deque.poll());
                }
                //将[出栈
                deque.pop();
                //[]中的字母
                String letter = builder.toString();
                //将buffer清空
                builder = new StringBuilder();
                //将[]左边的数字出栈
                while (!deque.isEmpty() && Character.isDigit(deque.peek().charAt(0))) {
                    builder.insert(0, deque.poll());
                }
                //[]中字母重复的次数
                int num = Integer.parseInt(builder.toString());
                //将buffer清空
                //将[]中的字母重复num遍后再入栈
                builder = new StringBuilder();
                for (int i = 0; i < num; i++) {
                    builder.append(letter);
                }
                deque.push(String.valueOf(builder));
            }
        }
        builder = new StringBuilder();
        while (!deque.isEmpty()) {
            builder.insert(0, deque.poll());
        }
        return builder.toString();
    }*/

}
